# 该文件的主要任务：通过flask组件，构建出路由+预测函数的应用

# 导包
from flask import Flask, request, jsonify
from llm_classify import get_deeepseek_res,get_zhipuai_res,get_qwen_res
import json
# 创建flask对象
app_aa = Flask(__name__)


#  创建预测接口（路由+预测函数）
@app_aa.route('/predict', methods=['POST','GET'])
def predict():
    # 获取用户请求中的数据
    data = request.get_json()
    print(f'data:{data},{type(data)}')

    # 调用预测函数
    result = get_deeepseek_res(data["text"])
    # print(result)

    # 解析LLM返回的JSON字符串
    try:
        result_json = json.loads(result)
        # 返回结果
        return jsonify(result_json)
    except json.JSONDecodeError:
        # 如果解析失败，返回错误信息
        return jsonify({"error": "解析结果失败", "raw_result": result}), 500


# 启动app应用
if __name__ == '__main__':
    app_aa.run(host='192.168.109.45', port=11111, debug=True)
